import sys

def generate_cname(src_filename, dst_filename):
    # 读取输入文件中的域名列表
    with open(src_filename, 'r') as input_file:
        input_domains = input_file.read().strip().split('\n')

    # 处理每个域名并生成结果
    result = []
    for domain in input_domains:
        result.append(f"{domain} CNAME .")
        result.append(f"*.{domain} CNAME .")

    # 将结果写入输出文件
    with open(dst_filename, 'w') as output_file:
        output_file.write('\n'.join(result))

    print(f"结果已写入 {dst_filename}")

if __name__ == "__main__":
    if len(sys.argv) != 3:
        print("用法: python3 generate_cname.py <源文件> <目标文件>")
        sys.exit(1)

    src_file = sys.argv[1]
    dst_file = sys.argv[2]

    generate_cname(src_file, dst_file)
